//
//  WBTitleButton.swift
//  DXWeibo
//
//  Created by dx on 2017/1/11.
//  Copyright © 2017年 dx. All rights reserved.
//

import UIKit

class WBTitleButton: UIButton {

    /// 重载构造函数
    ///
    /// - Parameter title: title 是动的，所以参数传 title
    
    // - title 如果是 nil，就显示“首页”
    // - 如果不为 nil，显示 title 和 箭头图像
    init(title: String?) {
        super.init(frame: CGRect())

        // 1> 判断 title 是否为 nil
        if title == nil {
            setTitle("首页", for: .normal)
        } else {
            setTitle(title! + " ", for: .normal)
            
            // 设置图像
            setImage(UIImage(named: "navigationbar_arrow_down"), for: .normal)
            setImage(UIImage(named: "navigationbar_arrow_up"), for: .selected)
        }
        
        // 2> 设置字体和颜色
        titleLabel?.font = UIFont.boldSystemFont(ofSize: 17)
        setTitleColor(UIColor.darkGray, for: .normal)
        
        // 3> 设置大小
        sizeToFit()
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    /// 重新布局子视图
    override func layoutSubviews() {
        super.layoutSubviews()
        
        // 判断 label 和 imageView 是否同时存在
        guard let titleLabel = titleLabel,
            let imageView = imageView else {
                return
        }
        
        // 改变子控件位移
        // OC 中不允许直接修改结构体内部的值，Swift 中可以
        titleLabel.frame.origin.x = 0
        imageView.frame.origin.x = titleLabel.bounds.width
    }
}
















